"""
配置文件
包含API密钥、模型参数等配置信息
"""

import os
from dotenv import load_dotenv

# 加载环境变量
load_dotenv()

class Config:
    """配置类"""
    
    # API配置
    BASE_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1"
    API_KEY = os.getenv("DASHSCOPE_API_KEY", "sk-d86c4e3a4aca43769e6ac6edfbcb4cc2")
    
    # 模型配置
    TEXT_MODEL = "qwen3-235b-a22b"
    VISION_MODEL = "qwen-vl-plus"
    EMBEDDING_MODEL = "text-embedding-v4"
    
    # 向量数据库配置
    CHROMA_HOST = "localhost"
    CHROMA_PORT = 8000
    
    # 数据处理配置
    IMAGE_SIZE = (224, 224)  # 图像标准化尺寸
    MAX_TEXT_LENGTH = 512    # 文本最大长度
    
    # 分类配置
    NUM_CLASSES = 10         # 目标类别数量
    SIMILARITY_THRESHOLD = 0.7  # 相似度阈值
    
    # 可视化配置
    FIGURE_SIZE = (12, 8)
    DPI = 100
    
    # 文件路径配置
    DATA_DIR = "data"
    INPUT_DIR = "data/input"
    OUTPUT_DIR = "data/output"
    MODEL_DIR = "models"
    
    # 日志配置
    LOG_LEVEL = "INFO"
    LOG_FILE = "app.log" 